package tetris.features;

import java.util.ArrayList;
import tetris.simulator.State;

public class ColHeightFeature extends Feature{
  
  public ArrayList<Double> getFeatureVector(State tetrisState) {
    ArrayList<Double> featureVector = new ArrayList<Double>();
    for (int i = 0; i < tetrisState.getTop().length; i++)
      featureVector.add(new Double(tetrisState.getTop()[i])); 
    return featureVector;
  }
}
